python animation to get colored square spiral